/*
 * Copyright 2015 GenerallyCloud.com
 *  
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *  
 *      http://www.apache.org/licenses/LICENSE-2.0
 *  
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */ 
package com.generallycloud.nio.codec.protobuf.future;

import java.util.HashMap;
import java.util.Map;

import com.generallycloud.nio.component.IoEventHandleAdaptor;
import com.google.protobuf.InvalidProtocolBufferException;
import com.google.protobuf.MessageLite;
import com.google.protobuf.Parser;

public abstract class ProtobufIOEventHandle extends IoEventHandleAdaptor {

	private Map<String, Parser<? extends MessageLite>>	parses	= new HashMap<String, Parser<? extends MessageLite>>();

	public void regist(String name, Parser<? extends MessageLite> parser) {
		parses.put(name, parser);
	}

	public void regist(MessageLite messageLite) {

		parses.put(messageLite.getClass().getName(), messageLite.getParserForType());
	}

	public Parser<? extends MessageLite> getParser(String name) throws InvalidProtocolBufferException {

		Parser<? extends MessageLite> parser = parses.get(name);

		if (parser == null) {
			throw new InvalidProtocolBufferException("did not found parse by name " + name);
		}

		return parser;
	}

}
